import os
from datetime import datetime

import pytz

def now_time_str():
    """获取当前时间"""
    dt = now_time()
    # time_str = dt.strftime("%Y年%m月%d日%H时%M分%S秒")
    time_str = dt.strftime("%Y{Y}%m{m}%d{d} %H{H}%M{M}%S{S}").format(Y='年', m='月', d='日', H='时', M='分', S='秒')
    return time_str

def now_time():
    """
    获取当前时间
    如果设置了 TZ , datetime.now() 时候获取到的是UTC时间
        此函数能够获得正确的时间
    """
    if 'TZ' in os.environ:
        shanghai_tz = pytz.timezone(os.environ["TZ"])
        now = datetime.now(shanghai_tz)
        return now
    return datetime.now()